package com.tianji.promotion.service.impl;

import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.tianji.common.domain.dto.PageDTO;
import com.tianji.common.utils.CollUtils;
import com.tianji.promotion.constans.PromotionConstants;
import com.tianji.promotion.domain.po.Coupon;
import com.tianji.promotion.domain.po.ExchangeCode;
import com.tianji.promotion.domain.query.CodeQuery;
import com.tianji.promotion.domain.vo.ExchangeCodeVO;
import com.tianji.promotion.mapper.ExchangeCodeMapper;
import com.tianji.promotion.service.IExchangeCodeService;
import com.tianji.promotion.utils.CodeUtil;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Service;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import static com.tianji.promotion.constans.PromotionConstants.COUPON_CODE_MAP_KEY;
import static com.tianji.promotion.constans.PromotionConstants.COUPON_RANGE_KEY;

/**
 * <p>
 * 兑换码 服务实现类
 * </p>
 *
 * @author zr
 * @since 2024-07-30
 */
@Service
@Slf4j
@RequiredArgsConstructor
public class ExchangeCodeServiceImpl extends ServiceImpl<ExchangeCodeMapper, ExchangeCode> implements IExchangeCodeService {

    private final StringRedisTemplate redisTemplate;
    /**
     * 异步生成优惠券兑换码。
     * <p>
     * 此方法设计为异步执行，目的是为了不阻塞调用线程，提高系统的并发处理能力。
     *
     * @param coupon 优惠券对象，包含生成代码所需的信息。
     *               之所以需要传入优惠券对象，是因为生成代码可能需要基于优惠券的特定属性。
     */
    @Override
    @Async("generateExchangeCodeExecutor")
    public void asyncGenerateCode(Coupon coupon) {
        log.debug("异步执行生成兑换码，线程id{}; 线程名称{}", Thread.currentThread().getId(), Thread.currentThread().getName());
        // 发放数量
        Integer totalNum = coupon.getTotalNum();
        // 1.获取Redis自增序列号
        Long increment = redisTemplate.opsForValue().increment(PromotionConstants.COUPON_CODE_SERIAL_KEY, totalNum);
        if(increment == null){
            return;
        }
        //2.使用工具类，批量生成兑换码，由于拿到的是最新的id，所以我们要计算出第一个id是多少
        int startId = increment.intValue() - totalNum + 1;
        List<ExchangeCode> list = new ArrayList<>(totalNum);
        for (int serialId = startId; serialId <= increment.intValue(); serialId++) {
            //用兑换码的id+新鲜值，基于类base32的方式来生成一个唯一的兑换码
            //参数1：兑换码的id，参数2：新鲜值生成所需参数()，这里直接用优惠券的id，工具会生成0-15的数字，代表16个密钥
            String code = CodeUtil.generateCode(serialId, coupon.getId());
            ExchangeCode exchangeCode = new ExchangeCode();
            exchangeCode.setCode(code);
            exchangeCode.setId(serialId); //手动生成的兑换码的id, 该po类的注解应该为INPUT
            exchangeCode.setExchangeTargetId(coupon.getId());
            exchangeCode.setExpiredTime(coupon.getIssueEndTime());
            list.add(exchangeCode);
        }
        // 3.保存数据库
        saveBatch(list);
        // 4.写入Redis缓存，member：couponId，score：兑换码的最大序列号
        redisTemplate.opsForZSet().add(COUPON_RANGE_KEY, coupon.getId().toString(), increment);
    }

    /**
     * 查询兑换码分页列表
     * <p>
     * 此方法主要用于后台管理系统对兑换码的分页查询，以便更高效地处理大量兑换码信息
     * 它根据传入的查询参数，返回满足条件的兑换码分页数据
     *
     * @param query 兑换码的查询条件，包括页码、每页数量等分页信息，以及可能的筛选条件
     * @return 返回一个PageDTO对象，包含满足条件的兑换码分页数据
     */
    @Override
    public PageDTO<ExchangeCodeVO> queryCodePage(CodeQuery query) {
        // 1.分页查询兑换码
        Page<ExchangeCode> page = lambdaQuery()
                .eq(ExchangeCode::getStatus, query.getStatus())
                .eq(ExchangeCode::getExchangeTargetId, query.getCouponId())
                .page(query.toMpPage());
        // 2.返回数据
        return PageDTO.of(page, c -> new ExchangeCodeVO(c.getId(), c.getCode()));
    }

    /**
     * 验证兑换码是否兑换
     *
     * @param serialNum 兑换码id
     * @param mark      标记已兑换
     * @return 是否已经兑换
     */
    @Override
    public boolean updateExchangeMark(long serialNum, boolean mark) {
        Boolean boo = redisTemplate.opsForValue().setBit(COUPON_CODE_MAP_KEY, serialNum, mark);
        return boo != null && boo;
    }

    @Override
    public Long exchangeTargetId(long serialNum) {
        // 1.查询score值比当前序列号大的第一个优惠券
        Set<String> results = redisTemplate.opsForZSet().rangeByScore(
                COUPON_RANGE_KEY, serialNum, serialNum + 5000, 0L, 1L);
        if (CollUtils.isEmpty(results)) {
            return null;
        }
        // 2.数据转换
        String next = results.iterator().next();
        return Long.parseLong(next);
    }
}
